#!/bin/bash

# 开始部署
echo "开始部署 Ollama 并运行模型..."

# 检查 Ollama 是否已安装
if command -v ollama &> /dev/null; then
    echo "Ollama 已安装。"
else
    echo "Ollama 未安装，正在安装 Ollama..."
    curl -fsSL https://ollama.com/install.sh | sh
    if [ $? -ne 0 ]; then
        echo "安装 Ollama 失败！"
        exit 1
    fi
fi

# 检查 Ollama 版本
ollama --version
if [ $? -ne 0 ]; then
    echo "Ollama 安装异常，无法获取版本号。"
    exit 1
fi

# 启动 Ollama 服务
echo "正在启动 Ollama 服务..."
ollama serve &
OLLAMA_PID=$!

# 等待几秒确保服务启动
sleep 5

# 检查服务是否启动成功
if ps -p $OLLAMA_PID > /dev/null; then
    echo "Ollama 服务已启动，PID: $OLLAMA_PID"
else
    echo "Ollama 服务启动失败。可能端口已被占用！"
    exit 1
fi

# 运行模型
MODEL_NAME="qwen3:1.7b"
echo "正在运行模型: $MODEL_NAME"
ollama run "$MODEL_NAME"

if [ $? -eq 0 ]; then
    echo "模型运行成功！"
    kill $OLLAMA_PID &> /dev/null
    exit 0
else
    echo "模型运行失败！"
    kill $OLLAMA_PID &> /dev/null
    exit 1
fi